from enum import Enum


class AccountEnum(Enum):
    # ASSETS_CURRENT_BANK_CMB1234 = 'Assets:Current:Bank:CMB1234'
    # ASSETS_CURRENT_BANK_ABC5770 = 'Assets:Current:Bank:ABC1234'
    ASSETS_CURRENT_WECHAT_MINIFUND = 'Assets:Investments:Wechat:零钱通'
    # ASSETS_CURRENT_WECHAT_WALLET = 'Assets:Current:Wechat:Wallet'

    EXPENSES_FOOD_VT = 'Expenses:Food:买菜'
    EXPENSES_FOOD_OB = 'Expenses:Food:外卖'
    EXPENSES_FOOD_FT = 'Expenses:Food:水果'
    EXPENSES_TRANSPORT = 'Expenses:Transport'
    EXPENSES_CLOTHING = 'Expenses:Clothing'
    EXPENSES_OTHER = 'Expenses:Other'
    EXPENSES_STUDY = 'Expenses:Study'
    EXPENSES_CONSUMPTION = 'Expenses:Consumption'
    EXPENSES_KYLE = 'Expenses:Kyle'

    LIABILITIES_CMB = 'Liabilities:CreditCard:招商'
    LIABILITIES_GF = 'Liabilities:CreditCard:广发'
    LIABILITIES_PA = 'Liabilities:CreditCard:平安'

    INCOME_OTHER = 'Income:Other'

    
    def __str__(self):
        return self.value


class ProviderEnum(Enum):
    WECHAT = 'wechat'
    # ALIPAY = 'alipay'

    def __str__(self):
        return self.value


class WechatColumnEnum(Enum):
    TRADE_TYPE = 1  # 交易类型
    TRADE_OBJECT = 2  # 交易对方
    PRODUCT = 3  # 商品
    INCOME_EXPENSE = 4  # 收/支
    PAY_METHOD = 6  # 支付方式
    PAY_STATUS = 7  # 当前状态


class AlipayColumnEnum(Enum):
    TRADE_TYPE = 1
    TRADE_OBJECT = 2
    PRODUCT = 4
    INCOME_EXPENSE = 5
    PAY_METHOD = 7
    PAY_STATUS = 8